package com.heima.schedule.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.CollectionUtils;
import com.heima.common.constants.RedisConstants;
import com.heima.common.constants.ScheduleConstants;
import com.heima.model.schedule.dtos.Task;
import com.heima.model.schedule.pojos.Taskinfo;
import com.heima.model.schedule.pojos.TaskinfoLogs;
import com.heima.schedule.mapper.TaskinfoLogsMapper;
import com.heima.schedule.mapper.TaskinfoMapper;
import com.heima.schedule.service.TaskService;
import com.heima.utils.common.BeanHelper;
import com.heima.utils.common.JsonUtils;
import lombok.extern.slf4j.Slf4j;
import org.joda.time.DateTime;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.scheduling.annotation.Async;
import org.springframework.scheduling.annotation.Scheduled;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import java.util.ArrayList;
import java.util.Date;
import java.util.List;
import java.util.Set;

@Service
@Slf4j
public class TaskServiceImpl implements TaskService {

    @Autowired(required = false)
    private TaskinfoMapper taskinfoMapper;

    @Autowired(required = false)
    private TaskinfoLogsMapper taskinfoLogsMapper;

    @Autowired
    private StringRedisTemplate redisTemplate;

    @Override
    @Transactional
    public long addTask(Task task) {
        //把任务信息存入DB
        addTaskToDB(task);

        //把任务信息存入Redis
        addTaskToCache(task);

        //返回新增的任务ID
        return task.getTaskId();
    }

    private void addTaskToCache(Task task) {
        //判断任务的执行时间是否在未来5分钟以内
        //方式一：task的执行时间的毫秒值-当前时间毫秒值 < 5*60*1000
        /**
         * plusMinutes(): +分钟
         * minusMinutes(): -分钟
         */
        long futureTime = DateTime.now().plusMinutes(5).getMillis();//5分钟后的时间
        if (task.getExecuteTime() <= futureTime) {
            // 为什么加上taskTopic，因为一种类型的任务存储一份ZSet，如文章定时发布，task_1 , 订单取消发布：task_2 .....
            String key= RedisConstants.TASK_KEY+task.getTaskTopic();
            //score存储的就是该任务的执行时间
            redisTemplate.opsForZSet().add(key, JsonUtils.toString(task),task.getExecuteTime());
        }
    }

    private void addTaskToDB(Task task) {
        try {
            Taskinfo taskinfo = BeanHelper.copyProperties(task, Taskinfo.class);
            taskinfo.setExecuteTime(new Date(task.getExecuteTime()));

            //存入任务表
            taskinfoMapper.insert(taskinfo);

            //存入任务日志表
            TaskinfoLogs taskinfoLogs = BeanHelper.copyProperties(taskinfo, TaskinfoLogs.class);
            taskinfoLogs.setVersion(1);//设置初始版本
            taskinfoLogs.setStatus(ScheduleConstants.SCHEDULED);//待执行状态
            taskinfoLogsMapper.insert(taskinfoLogs);

            //手动给task的任务ID赋值
            task.setTaskId(taskinfo.getTaskId());
        } catch (Exception e) {
            log.error("存储任务失败：{}",e.getMessage());
            throw new RuntimeException(e);
        }
    }

    @Override
    public List<Task> pollTask(Integer taskTopic) {
        String key= RedisConstants.TASK_KEY+taskTopic;
        //查询redis的zset，取出满足条件的任务列表
        Set<String> taskSet = redisTemplate.opsForZSet().rangeByScore(key, 0, DateTime.now().getMillis());

        List<Task> taskList = new ArrayList<>();
        if (CollectionUtils.isNotEmpty(taskSet)) {
            for (String taskJson : taskSet) {
                Task task = JsonUtils.toBean(taskJson, Task.class);

                //删除任务表的任务，修改任务日志表的status为2
                updateTaskToDB(task);

                //删除缓存任务
                redisTemplate.opsForZSet().remove(key,taskJson);

                taskList.add(task);
            }
        }

        return taskList;
    }

    private void updateTaskToDB(Task task) {
        try {
            //删除任务表的任务
            taskinfoMapper.deleteById(task.getTaskId());

            // 修改任务日志表的status为2
            TaskinfoLogs taskinfoLogs = taskinfoLogsMapper.selectById(task.getTaskId());
            taskinfoLogs.setStatus(ScheduleConstants.EXECUTED); //1:代表任务被执行
            taskinfoLogsMapper.updateById(taskinfoLogs);
        } catch (Exception e) {
            log.error("删除任务失败：{}",e.getMessage());
            throw new RuntimeException(e);
        }
    }

    /**
     * 每隔5分钟定时同步MySQL任务到缓存  (0 0/5 * * * ?)
     *  cron格式： 秒 分 时 日 月 周
     */
    @Scheduled(cron = "0/30 * * * * ?")
    public void syncTaskToCache(){
        //查询出未来5分钟内执行的任务，才需要同步到redis中
        Date futureDate = DateTime.now().plusMinutes(5).toDate();

        QueryWrapper<Taskinfo> queryWrapper = new QueryWrapper<>();
        queryWrapper.le("execute_time",futureDate);
        List<Taskinfo> taskinfoList = taskinfoMapper.selectList(queryWrapper);

        if (CollectionUtils.isNotEmpty(taskinfoList)) {
            for (Taskinfo taskinfo : taskinfoList) {
                Task task = BeanHelper.copyProperties(taskinfo,Task.class);
                task.setExecuteTime(taskinfo.getExecuteTime().getTime());
                addTaskToCache(task);
            }
        }
        log.info("同步数据完成");
    }
}
